package com.gitee.deament.server.core.cli.db.set;

import com.gitee.deament.server.config.SystemConfig;
import com.gitee.deament.server.core.annotation.CliHander;
import com.gitee.deament.server.core.cli.AbstractCli;
import com.gitee.deament.server.core.data.ConnectData;
import com.gitee.deament.server.core.data.DBData;
import com.gitee.deament.server.core.data.datatype.Value;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;

import java.util.ArrayList;

import java.util.List;
import java.util.Set;

/**
 * SREM key member [member …]
 * 可用版本： >= 1.0.0
 * 时间复杂度: O(N)， N 为给定 member 元素的数量。
 * 移除集合 key 中的一个或多个 member 元素，不存在的 member 元素会被忽略。
 *
 * 当 key 不是集合类型，返回一个错误。
 *
 * Note
 *
 * 在 Redis 2.4 版本以前， SREM 只接受单个 member 值。
 *
 * 返回值
 * 被成功移除的元素的数量，不包括被忽略的元素。
 * @author deament
 */
@CliHander(command = "SREM")
public class SREMcli extends AbstractCli {
    @Override
    public void executeCli(ChannelHandlerContext channelHandlerContext, String command) {
        List<String> cliValues = getValues(command);
        if (cliValues.size() <3) {
            channelHandlerContext.channel().writeAndFlush(new TextWebSocketFrame("ERR wrong number of arguments for 'SREM' command"));
            return;
        }
        int db = ConnectData.SELECT_DB_STORAGE.getUseDB(channelHandlerContext.channel().id().asLongText());
        DBData dbData = SystemConfig.dbData;
        String key = cliValues.get(1);
        List<String> setMemebrs=new ArrayList<>();
        for(int i=2;i<cliValues.size();i++){
            setMemebrs.add(cliValues.get(i));
        }
        try {
            Value<Set<String>> set = dbData.getSet(db, key);
            if (set == null || set.getValue().size() == 0) {
                channelHandlerContext.channel().writeAndFlush(new TextWebSocketFrame("nil"));
                return;
            }
            int removeCount=0;
            for(String member:setMemebrs){
              boolean removeResult=  set.getValue().remove(member);
              if(removeResult){
                  removeCount++;
              }
            }
            channelHandlerContext.channel().writeAndFlush(new TextWebSocketFrame(removeCount+""));



        } catch (Exception e) {
            channelHandlerContext.channel().writeAndFlush(new TextWebSocketFrame(e.getMessage()));
        }

    }
}
